"""
* 名称：csv自动在最左侧添加一列序号.py
* 功能： pytorch需要第一列数据是0-x的序号，才能正常识别数据集
* 注意：
"""

import pandas as pd


def add_sequence_num(file: str, re_file: str):
    # 读取csv文件
    df = pd.read_csv(f'./dataset/{file}')

    # 创建一个新的列，这个列的值从0开始递增
    df.insert(0, 0, range(1, 1 + len(df)))

    # 保存修改后的csv文件
    df.to_csv(f'./dataset/{re_file}', index = False)

    print("done")
